# main.py
from user import NormalUser, VIPUser, GoldUser, SilverUser
from price_strategy import NormalUserPrice, VIPUserPrice, GoldUserPrice, SilverUserPrice

def main():
    # 创建普通用户，并为其注入全价策略
    normal_user = NormalUser("张三", NormalUserPrice())
    normal_user.display()
    print(f"价格：{normal_user.calculate_price(100)}")  # 计算价格
    
    # 创建VIP用户，并为其注入VIP用户折扣策略
    vip_user = VIPUser("李四", VIPUserPrice())
    vip_user.display()
    print(f"价格：{vip_user.calculate_price(100)}")  # 计算价格

    # 创建金卡用户，并为其注入金卡折扣策略
    gold_user = GoldUser("王五", GoldUserPrice())
    gold_user.display()
    print(f"价格：{gold_user.calculate_price(100)}")  # 计算价格

    # 创建银卡用户，并为其注入银卡折扣策略
    silver_user = SilverUser("赵六", SilverUserPrice())
    silver_user.display()
    print(f"价格：{silver_user.calculate_price(100)}")  # 计算价格

if __name__ == "__main__":
    main()

